package com.xxx.rpc.protocol;

import com.xxx.rpc.config.Config;
import com.xxx.rpc.message.Message;
import io.netty.buffer.ByteBuf;
import io.netty.channel.ChannelHandler;
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.MessageToMessageCodec;
import lombok.extern.slf4j.Slf4j;

import java.util.List;

/**
 * 消息编码和解码器
 * 继承MessageToMessageCodec
 * 必须和 LengthFieldBasedFrameDecoder 一起使用，确保接到的 ByteBuf 消息是完整的，解决了半包和粘包问题
 */
@Slf4j
@ChannelHandler.Sharable
public class MessageCodecSharable extends MessageToMessageCodec<ByteBuf, Message> {

    // 编码器
    @Override
    public void encode(ChannelHandlerContext ctx, Message msg, List<Object> outList) throws Exception {
        ByteBuf out = ctx.alloc().buffer();
        // 1. 4字节的魔数
        out.writeBytes(new byte[]{1, 2, 3, 4});
        // 2. 1字节的版本,
        out.writeByte(1);
        // 3. 1字节的序列化方式 jdk 0 , json 1，枚举的ordinal()方法表示该枚举成员在枚举类里声明的序号
        out.writeByte(Config.getSerializerAlgorithm().ordinal());
        // 4. 1字节的指令类型
        out.writeByte(msg.getMessageType());
        // 5. 4个字节请求序号
        out.writeInt(msg.getSequenceId());
        // 无意义，1个字节对齐填充：1111 1111
        out.writeByte(0xff);
        // 6. 获取内容的字节数组，这里可以自定义序列化算法（通过Config读取配置文件来指定）
        byte[] bytes = Config.getSerializerAlgorithm().serialize(msg);
        // 7. 4个字节消息长度
        out.writeInt(bytes.length);
        // 8. 写入消息正文内容
        out.writeBytes(bytes);
        outList.add(out);
    }

    // 解码器
    @Override
    protected void decode(ChannelHandlerContext ctx, ByteBuf in, List<Object> out) throws Exception {
        int magicNum = in.readInt();    // 4字节魔数
        byte version = in.readByte();   // 1字节版本号
        byte serializerAlgorithm = in.readByte(); // 1字节序列化算法：0 或 1
        byte messageType = in.readByte(); // 1字节消息类型：0,1,2...
        int sequenceId = in.readInt();  // 4字节请求序号
        in.readByte();  // 1字节无意义填充
        int length = in.readInt();  // 4字节消息长度
        byte[] bytes = new byte[length];
        in.readBytes(bytes, 0, length);  // 读取[0, length]范围内的字节到bytes数组里作为消息正文

        // 找到反序列化算法，这里利用枚举的序列化算法序号找到对应的算法
        Serializer.Algorithm algorithm = Serializer.Algorithm.values()[serializerAlgorithm];
        // 确定具体消息类型class
        Class<? extends Message> messageClass = Message.getMessageClass(messageType);
        Message message = algorithm.deserialize(messageClass, bytes);
//        log.debug("{}, {}, {}, {}, {}, {}", magicNum, version, serializerType, messageType, sequenceId, length);
//        log.debug("{}", message);
        out.add(message);
    }

}
